package com.zsr.rabbitmq.four;

import com.rabbitmq.client.Channel;
import com.rabbitmq.client.ConfirmCallback;
import com.zsr.rabbitmq.utils.RabbitMqUtils;

import java.util.Scanner;
import java.util.UUID;
import java.util.concurrent.ConcurrentNavigableMap;
import java.util.concurrent.ConcurrentSkipListMap;

/**
 * @author zsr
 * @date 2021/8/5 17:21
 *      发布确认模式
 *      1.单个确认
 *      2.批量确认
 *      3.异步批量确认
 */
public class ConfirmMessage {

    public static final int MESSAGE_COUNT= 1000;

    public static void main(String[] args) throws Exception{
        //1.单个确认
        //publishMessageIndividually();//发送1000个消息共花费597ms时间
        //2.批量确认
        //publishMessageBatch();//发送1000个,批量确认消息共花费70ms时间
        //3.异步确认
        publishMessageAsync();//发送1000个,异步确认消息共花费30ms时间
                              //发送1000个,异步确认消息共花费37ms时间
    }

    /**
     * 单个确认
     * @throws Exception
     */
    public static void publishMessageIndividually()throws Exception{
        Channel channel= RabbitMqUtils.getChannel();
        String queueName = UUID.randomUUID().toString();//36位长度
        //队列名  是否持久化 是否共享  是否自动删除  其他参数
        channel.queueDeclare(queueName,false,false,false,null);
        //开启确认模式
        channel.confirmSelect();
        long begin =System.currentTimeMillis();

        //MESSAGE_COUNT.fori 快捷键
        for (int i = 0; i < MESSAGE_COUNT; i++) {
            channel.basicPublish("",queueName,false,null,(i+"").getBytes());
            //----------------单个消息马上进行确认----------------------
            boolean flag = channel.waitForConfirms();//等待确认结果
            if(flag) System.out.println("消息发送成功");
        }
        //结束时间
        long end=System.currentTimeMillis();
        System.out.println("发送"+MESSAGE_COUNT+"个消息共花费"+(end-begin)+"ms时间");
    }

    /**
     * 批量发布确认
     * @throws Exception
     */
    public static void publishMessageBatch()throws Exception{
        Channel channel = RabbitMqUtils.getChannel();
        String queueName = UUID.randomUUID().toString();//36位长度
        //队列名  是否持久化 是否共享  是否自动删除  其他参数
        channel.queueDeclare(queueName,false,false,false,null);
        //开启确认模式
        channel.confirmSelect();
        //开始时间
        long begin =System.currentTimeMillis();
        //批量确认消息大小 (比如每一百一百条的确认)
        int batchSize =100;
        //批量发送数据 批量发布确认
        for (int i = 0; i < MESSAGE_COUNT; i++) {
            channel.basicPublish("",queueName,false,null,(i+"").getBytes());
            //----------------批量消息进行确认----------------------
            if((i+1)%batchSize==0){
                channel.waitForConfirms();
            }
        }

        //结束时间
        long end=System.currentTimeMillis();
        System.out.println("发送"+MESSAGE_COUNT+"个,批量确认消息共花费"+(end-begin)+"ms时间");

    }

    /**
     * 异步发布确认
     * @throws Exception
     */
    public static void publishMessageAsync()throws Exception{
        Channel channel = RabbitMqUtils.getChannel();
        String queueName = UUID.randomUUID().toString();//36位长度
        //队列名  是否持久化 是否共享  是否自动删除  其他参数
        channel.queueDeclare(queueName,false,false,false,null);
        //开启确认模式
        channel.confirmSelect();
        /**
         * 创建一个线程安全有序的一个哈希表 适用于高并发的情况下
         * 用途: 1.轻松的将序号和消息进行关联
         *      2.轻松批量删除条目 只要给到序号
         *      3.支持高并发
         */
        ConcurrentSkipListMap<Long,String> concurrentSkipListMap=
                new ConcurrentSkipListMap<>();

        //----------------准备消息的监听器 监听哪些消息成功 哪些消息失败----------------------
        /**
         * 成功回调
         * 1. 消息的标记
         * 2. 是否为批量确认
         */
        ConfirmCallback ackCallback =(deliveryTag,multiple)->{
            //2.删除掉已经确认的消息 剩下的就是未确认的消息
            if(multiple){//如果是批量 批量清除map
                //ConcurrentSkipListMap是ConcurrentNavigableMap接口的实现类
                //headMap()返回此地图部分的视图，其密钥严格小于 toKey 。
                ConcurrentNavigableMap<Long, String> confirmed =
                        concurrentSkipListMap.headMap(deliveryTag);
                //System.out.println(confirmed);
                confirmed.clear();
            }else{
                concurrentSkipListMap.remove(deliveryTag);
            }
            System.out.println("确认的消息的编号:"+deliveryTag);
        };
        /**
         * 失败回调
         * 1. 消息的标记
         * 2. 是否为批量确认
         */
        ConfirmCallback nackCallback=(deliveryTag,multiple)->{
            System.out.println("为确认的消息的编号:"+deliveryTag);
        };
        //监听器 不需要监听填null (异步通知)
        channel.addConfirmListener(ackCallback,nackCallback);

        //开始时间
        long begin =System.currentTimeMillis();

        for (int i = 0; i < MESSAGE_COUNT; i++) {
            channel.basicPublish("",queueName,false,null,(i+"").getBytes());
            /**
             * 如何处理异步未确认的消息
             * 1.在此处记录下所有要发送的消息 消息的总和
             * 2.在发布成功回调函数处，删除掉已经确认的消息 剩下的就是未确认的消息
             * */
            concurrentSkipListMap.put(channel.getNextPublishSeqNo(),i+"");
        }
        
        //结束时间
        long end=System.currentTimeMillis();
        System.out.println("发送"+MESSAGE_COUNT+"个,异步确认消息共花费"+(end-begin)+"ms时间");

    }
}
